import time
from selenium import webdriver
from selenium.webdriver.common.by import By

# 1. 初始化Chrome浏览器驱动
driver = webdriver.Chrome(executable_path='../01/chromedriver.exe')
driver.maximize_window()  # 浏览器窗口最大化

# 2. 访问目标页面
driver.get("http://121.43.169.97:8848/pageA.html")

# 3. 填写注册信息：定位用户输入框并输入"admin"
driver.find_element(By.CSS_SELECTOR, "#userA").send_keys("admin")

# 4. 截图保存——方式1：固定文件名保存
driver.get_screenshot_as_file("../chr/admin.png")

# 方式2：用时间戳动态生成文件名保存
timestamp = time.strftime("%Y_%m_%d_%H_%M_%S")
driver.get_screenshot_as_file(f"../chr/{timestamp}.png")

# （可选）关闭浏览器，若需查看截图结果可注释此行
# driver.quit()